import requests
url = " http://localhost:11434/api/generate"

prompt ="""
你需要扮演资深Python代码审查员，
                生成包含真实缺陷的代码片段和对应的审查意见，遵循以下规则：
                1. 代码必须包含至少一个安全漏洞类问题
                2. 使用实际业务场景（如：API端点输入验证）
                3. 包含Python特性问题（如：[\'SQL注入风险\', \'XSS攻击\', \'不安全的反序列化\']）
                4. 输出格式严格遵循JSON，包含instruction,input,output三个键
                example:
                {
                    "instruction": "Analyze the following code and provide a review:",
                    "input": "code",
                    "output": "review"
                }
                注意：输出类型instruction,input,output必须是字符串类型，
                input类型为python代码，output类型为代码审查意见。
                除了代码之外的内容，要使用中文输出。
"""

data = {
    # "model": "deepseek-r1:8b",
    "model": "qwen2.5-coder:1.5b",
    "prompt": prompt,
    "stream": False,
    "format": "json",
    "temperature": 0.6
}
import json
response = requests.post(url, data=json.dumps(data))
data = json.loads(response.text)
result = data.get("response")
print(result)
# json_data = result.split("```json")[1].split("```")[0].strip()
# print(json.loads(json_data))

